/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | foam-extend: Open Source CFD
   \\    /   O peration     | Version:     4.1
    \\  /    A nd           | Web:         http://www.foam-extend.org
     \\/     M anipulation  | For copyright notice see file Copyright
-------------------------------------------------------------------------------
License
	This file is part of foam-extend.

	foam-extend is free software: you can redistribute it and/or modify it
	under the terms of the GNU General Public License as published by the
	Free Software Foundation, either version 3 of the License, or (at your
	option) any later version.

	foam-extend is distributed in the hope that it will be useful, but
	WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
	General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with foam-extend.  If not, see <http://www.gnu.org/licenses/>.

\*---------------------------------------------------------------------------*/

#include "primitiveMesh.H"

// * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //

void Foam::primitiveMesh::calcCells
(
	cellList& cellFaceAddr,
	const unallocLabelList& own,
	const unallocLabelList& nei,
	const label inNCells
)
{
	label nCells = inNCells;

	if (nCells == -1)
	{
		nCells = -1;

		forAll (own, faceI)
		{
			nCells = max(nCells, own[faceI]);
		}
		nCells++;
	}

	// 1. Count number of faces per cell

	labelList ncf(nCells, 0);

	forAll (own, faceI)
	{
		ncf[own[faceI]]++;
	}

	forAll (nei, faceI)
	{
		if (nei[faceI] >= 0)
		{
			ncf[nei[faceI]]++;
		}
	}

	// Create the storage
	cellFaceAddr.setSize(ncf.size());


	// 2. Size and fill cellFaceAddr

	forAll (cellFaceAddr, cellI)
	{
		cellFaceAddr[cellI].setSize(ncf[cellI]);
	}
	ncf = 0;

	forAll (own, faceI)
	{
		label cellI = own[faceI];

		cellFaceAddr[cellI][ncf[cellI]++] = faceI;
	}

	forAll (nei, faceI)
	{
		label cellI = nei[faceI];

		if (cellI >= 0)
		{
			cellFaceAddr[cellI][ncf[cellI]++] = faceI;
		}
	}
}


void Foam::primitiveMesh::calcCells() const
{
	// Loop through faceCells and mark up neighbours

	if (debug)
	{
		Pout<< "primitiveMesh::calcCells() : calculating cells"
			<< endl;
	}

	// It is an error to attempt to recalculate cells
	// if the pointer is already set
	if (cfPtr_)
	{
		FatalErrorIn("primitiveMesh::calcCells() const")
			<< "cells already calculated"
			<< abort(FatalError);
	}
	else
	{
		// Create the storage
		cfPtr_ = new cellList(nCells());
		cellList& cellFaceAddr = *cfPtr_;

		calcCells
		(
			cellFaceAddr,
			faceOwner(),
			faceNeighbour(),
			nCells()
		);
	}
}


// * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //

const Foam::cellList& Foam::primitiveMesh::cells() const
{
	if (!cfPtr_)
	{
		calcCells();
	}

	return *cfPtr_;
}


// ************************************************************************* //
